Self-Improvement with Auxiliary Rewards: Learning from Verified Traces and Evaluating Transfer
辅助奖励驱动的自改进:从验证轨迹中学习并评估迁移
🧭 小模型推理自改进 · 轨迹级辅助奖励
Georgia Institute of Technology
提出用置信度/熵辅助轨迹偏好区分同等正确推理,提升小模型自改进与跨任务迁移
📌 工作参考价值(4 点)
- 业务 LLM 生成推荐理由/query/文案时,常只有点击/转化等弱反馈;可引入置信度(SC)或熵变化+局部平滑度(UID-Conj.)作为辅助奖励,对正反馈轨迹打分,筛选高质量样本做 SFT,在标注有限时提升迁移表现。
- 离线加权 SFT 比在线 TTRL 更稳定且证据更强:同一响应池按辅助奖励加权训练,较均匀加权提升 1.93 点(CI [0.44,3.44])。业务可优先做离线加权精调,成本低、便于上线。
- 辅助奖励选择依赖任务:SC 更适合判断指令遵循和轨迹质量;UID-Conj. 在多种训练设置中更利于下游收益。实际可两者都算,在验证集上选择权重或融合,而非只看推理偏好。
- 评估轨迹偏好应以最终业务指标为准,而非仅看推理期排序一致性;建立下游 A/B 或离线 proxy 评估,辅助奖励要暴露学习价值。